package com.ajin.hot100;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Code9 {
    public static void main(String[] args) {

    }

    /**
     * 定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。不考虑答案输出的顺序。
     * 示例 1:
     * 输入: s = "cbaebabacd", p = "abc"
     * 输出: [0,6]
     * 解释:
     * 起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
     * 起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。
     * 示例 2:
     * 输入: s = "abab", p = "ab"
     * 输出: [0,1,2]
     * 解释:
     * 起始索引等于 0 的子串是 "ab", 它是 "ab" 的异位词。
     * 起始索引等于 1 的子串是 "ba", 它是 "ab" 的异位词。
     * 起始索引等于 2 的子串是 "ab", 它是 "ab" 的异位词。
     *
     * @param s
     * @param p
     * @return
     */
    public static List<Integer> findAnagrams(String s, String p) {
        List<Integer> res = new ArrayList<>();
        if (s.length() < p.length()) return res;
        int[] pCount = new int[26];
        int[] sCount = new int[26];
        // 初始化频次数组
        for (int i = 0; i < p.length(); i++) {
            pCount[p.charAt(i) - 'a']++;
            sCount[s.charAt(i) - 'a']++;
        }
        if (Arrays.equals(pCount, sCount)) res.add(0);
        for (int i = p.length(); i < s.length(); i++) {
            // 加入一个新字符
            sCount[s.charAt(i) - 'a']++;
            // 移除滑动窗口最左边字符
            sCount[s.charAt(i - p.length()) - 'a']--;

            if (Arrays.equals(pCount, sCount)) {
                res.add(i - p.length() + 1);
            }
        }
        return res;
    }


    public List<Integer> findAnagrams2(String s, String p) {
        List<Integer> res = new ArrayList<>();
        int lenS = s.length(), lenP = p.length();
        if (lenS < lenP) return res;

        // 对 p 排序，作为比较标准
        char[] pArr = p.toCharArray();
        Arrays.sort(pArr);
        String sortedP = new String(pArr);

        // 枚举所有长度为 lenP 的子串
        for (int i = 0; i <= lenS - lenP; i++) {
            String sub = s.substring(i, i + lenP);
            char[] subArr = sub.toCharArray();
            Arrays.sort(subArr);
            String sortedSub = new String(subArr);

            if (sortedSub.equals(sortedP)) {
                res.add(i);
            }
        }

        return res;
    }

}
